package com.starhub.application.interfaces.function;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.service.IService;
import com.starhub.common.bean.agent.AiFunction;
import com.starhub.engine.ai.dto.AiFunctionDTO;
import com.starhub.engine.ai.vo.AiFunctionVO;
import dev.langchain4j.model.chat.ChatLanguageModel;

import java.util.List;
import java.util.Map;

/**
 * AI函数服务接口
 */
public interface IAiFunctionService extends IService<AiFunction> {

    /**
     * 新增AI函数
     *
     * @param functionDTO  函数DTO对象
     * @param operatorId   操作人ID
     * @param operatorName 操作人姓名
     * @return 是否成功
     */
    boolean addFunction(AiFunctionDTO functionDTO, String operatorId, String operatorName);

    /**
     * 更新AI函数
     *
     * @param functionDTO  函数DTO对象
     * @param operatorId   操作人ID
     * @param operatorName 操作人姓名
     * @return 是否成功
     */
    boolean updateFunction(AiFunctionDTO functionDTO, String operatorId, String operatorName);

    /**
     * 删除AI函数
     *
     * @param id 函数ID
     * @return 是否成功
     */
    boolean deleteFunction(Long id);

    /**
     * 获取AI函数详情
     *
     * @param id 函数ID
     * @return 函数VO对象
     */
    AiFunctionVO getFunctionDetail(Long id);

    /**
     * 分页查询AI函数列表
     *
     * @param page 页码
     * @param size 每页大小
     * @param name 函数名称，可为空
     * @param type 函数类型，可为空
     * @return 分页结果
     */
    IPage<AiFunctionVO> pageFunctions(int page, int size, String name, String type);

    /**
     * 启用函数
     *
     * @param id           函数ID
     * @param operatorId   操作人ID
     * @param operatorName 操作人姓名
     * @return 是否成功
     */
    boolean enableFunction(Long id, String operatorId, String operatorName);

    /**
     * 禁用函数
     *
     * @param id           函数ID
     * @param operatorId   操作人ID
     * @param operatorName 操作人姓名
     * @return 是否成功
     */
    boolean disableFunction(Long id, String operatorId, String operatorName);

    /**
     * 获取所有启用状态的AI函数工具规格
     * 
     * @return 工具规格列表
     */
    List<Object> getEnabledFunctionSpecifications();

    /**
     * 执行AI函数
     *
     * @param functionName 函数名称
     * @param arguments    函数参数
     * @return 执行结果
     */
    String executeFunction(String functionName, Map<String, Object> arguments);

    /**
     * 将消息和函数执行结果一起提交给模型，并获取新的回答
     * 
     * @param model        聊天语言模型
     * @param userMessage  用户消息
     * @param functionName 函数名称
     * @param arguments    函数参数
     * @param result       函数执行结果
     * @return 模型的新回答
     */
    String regenerateWithFunctionResult(
            ChatLanguageModel model,
            String userMessage,
            String functionName,
            Map<String, Object> arguments,
            String result);
}